package io.kiki.stack.http.soap;

import feign.Response;
import feign.codec.ErrorDecoder;

import javax.xml.soap.*;
import javax.xml.ws.soap.SOAPFaultException;
import java.io.IOException;

/**
 * Wraps the returned {@link SOAPFault} if present into a {@link SOAPFaultException}. So you need to
 * catch {@link SOAPFaultException} to retrieve the reason of the {@link SOAPFault}.
 *
 * <p>If no faults is returned then the default {@link ErrorDecoder} is used to return exception and
 * eventually retry the call.
 */
public class SOAPErrorDecoder implements ErrorDecoder {

    private final String soapProtocol;

    public SOAPErrorDecoder() {
        this.soapProtocol = SOAPConstants.DEFAULT_SOAP_PROTOCOL;
    }

    /**
     * SOAPErrorDecoder constructor allowing you to specify the SOAP protocol.
     *
     * @param soapProtocol a string constant representing the MessageFactory protocol.
     * @see SOAPConstants#SOAP_1_1_PROTOCOL
     * @see SOAPConstants#SOAP_1_2_PROTOCOL
     * @see SOAPConstants#DYNAMIC_SOAP_PROTOCOL
     * @see MessageFactory#newInstance(String)
     */
    public SOAPErrorDecoder(String soapProtocol) {
        this.soapProtocol = soapProtocol;
    }

    @Override
    public Exception decode(String methodKey, Response response) {
        if (response.body() == null || response.status() == 503)
            return defaultErrorDecoder(methodKey, response);

        SOAPMessage message;
        try {
            message = MessageFactory.newInstance(soapProtocol).createMessage(null, response.body().asInputStream());
            if (message.getSOAPBody() != null && message.getSOAPBody().hasFault()) {
                return new SOAPFaultException(message.getSOAPBody().getFault());
            }
        } catch (SOAPException | IOException e) {
            // ignored
        }
        return defaultErrorDecoder(methodKey, response);
    }

    private Exception defaultErrorDecoder(String methodKey, Response response) {
        return new ErrorDecoder.Default().decode(methodKey, response);
    }
}
